//
// Created by Administrator on 2021/6/9.
//

/*
集团里有 n 名员工，他们可以完成各种各样的工作创造利润。

第i种工作会产生profit[i]的利润，它要求group[i]名成员共同参与。如果成员参与了其中一项工作，就不能参与另一项工作。

工作的任何至少产生minProfit 利润的子集称为 盈利计划 。并且工作的成员总数最多为 n 。

有多少种计划可以选择？因为答案很大，所以 返回结果模10^9 + 7的值。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/profitable-schemes
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。*/

/**
* 动态规划
* 背包问题 有两个限制: 成员的总数(背包容量) 和 最小利润
 * 三维dp
 * dp[i][j][k]:表示在前 i 个工作中选择了 j 个员工，并且满足工作利润至少为 k 的情况下的盈利计划的总数目
 * 如果无法展开当前工作i，dp[i][j][k]=dp[i−1][j][k]
 * 如果可以展开当前工作i，dp[i][j][k]=dp[i−1][j][k]+dp[i−1][j−group[i]][max(0,k−profit[i])]
*/
#include <vector>
#include <iostream>

using namespace std;

class Solution {
public:
    int profitableSchemes(int n, int minProfit, vector<int> &group, vector<int> &profit) {
        int len = (int) group.size(), MOD = (int) 1e9 + 7;
        vector<vector<vector<int>>> dp(len + 1, vector<vector<int>>(n + 1, vector<int>(minProfit + 1, 0)));
        dp[0][0][0] = 1;
        for (int i = 1; i <= len; ++i) { // 遍历工作
            int members = group[i - 1], earn = profit[i - 1]; // 这项工作需要的人数和能获得的利润
            for (int j = 0; j <= n; j++) { // 遍历工作人数
                for (int k = 0; k <= minProfit; k++) { //
                    if (j < members) { // 无法开展任务 需要的人数太多
                        dp[i][j][k] = dp[i - 1][j][k];
                    } else { // 可以开展任务
                        dp[i][j][k] = dp[i - 1][j][k] + dp[i - 1][j - members][max(0, k - earn)];
                    }
                }
            }
        }
        int sum = 0;
        for (int j = 0; j <= n; j++) {
            sum = (sum + dp[len][j][minProfit]) % MOD;
        }
        return sum;
    }
};

class Solution2 {
public:
    int profitableSchemes(int n, int minProfit, vector<int> &group, vector<int> &profit) {
        // 二维简化
        vector<vector<int>> dp(n + 1, vector<int>(minProfit + 1, 0));
        for (int i = 0; i <= n; i++) {
            dp[i][0] = 1;
        }
        int len = (int) group.size(), MOD = (int) 1e9 + 7;
        for (int i = 1; i <= len; i++) {
            int members = group[i - 1], earn = profit[i - 1];
            for (int j = n; j >= members; j--) {
                for (int k = minProfit; k >= 0; k--) {
                    dp[j][k] = (dp[j][k] + dp[j - members][max(0, k - earn)]) % MOD;
                }
            }
        }
        return dp[n][minProfit];
    }
};


int main() {
    vector<int> group{2, 2};
    vector<int> profit{2, 3};
    Solution sol;
    cout << sol.profitableSchemes(5, 3, group, profit) << endl;
    return 0;
}